AI-based imaging analysis